from fastapi import APIRouter, Depends, Security
from fastapi.responses import JSONResponse

from myProfile.models.data.myProfile_models import Login, PsychoTest
from myProfile.repository.psychotests import PsychoTestRepository
from myProfile.models.request.myProfile_request import PsychoTestReq
from myProfile.db_config.odmantic_config import create_db_engine
from myProfile.security.secure import get_current_valid_user

router = APIRouter()


@router.post("/psychotests/add")
async def add_psychotests(req: PsychoTestReq,
                          current_user: Login = Security(
                              get_current_valid_user, scopes=["user"]),
                          engine=Depends(create_db_engine)):
    psychotest = PsychoTest(**req.model_dump())
    repo: PsychoTestRepository = PsychoTestRepository(engine)
    result = await repo.add_to_profile(current_user.id, psychotest)
    if result is True:
        return req
    else:
        return JSONResponse(
            content={"message": "add psychotests problem encountered."},
            status_code=500)


@router.delete("/psychotests/delete/{id}")
async def delete_psychotests(id: int, current_user: Login = Security(
                            get_current_valid_user, scopes=["user"]),
                          engine=Depends(create_db_engine)):
    repo: PsychoTestRepository = PsychoTestRepository(engine)
    result = await repo.remove_from_profile(current_user.id, id)
    if result is True:
        return JSONResponse(
            content={"message": "delete psychotests successfully"},
            status_code=201)
    else:
        return JSONResponse(
            content={"message": "delete psychotests unsuccessfully"},
            status_code=500)


@router.get("/psychotests/get")
async def get_psychotests(current_user: Login = Security(
                            get_current_valid_user, scopes=["user"]),
                          engine=Depends(create_db_engine)):
    repo: PsychoTestRepository = PsychoTestRepository(engine)
    return await repo.get_from_profile(current_user.id)
